package com.nny.demo141;

import com.nny.demo513.Emp;

import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;

public class Demo {
    public static void menu() {
        System.out.println("欢迎来到员工管理系统");
        System.out.println("0.退出系统");
        System.out.println("1.添加员工");
        System.out.println("2.查询所有员工");
        System.out.println("3.根据员工姓名查询");
        System.out.println("4.删除员工");
        System.out.println("5.根据员工编号查询");
        System.out.println("6.根据员工编号修改年龄");
        System.out.println("请选择你需要进行的操作系统");
    }

    public static void main(String[] args) {
        TestSet testSet = new TestSet();
        Set<com.nny.demo513.Emp> empSet = new HashSet<>();
        boolean flag = true;
        while (flag) {
            menu();
            Scanner scanner = new Scanner(System.in);
            int n = scanner.nextInt();
            switch (n) {
                case 0:
                    flag = false;
                    break;
                case 1:
                    System.out.println("请输入员工信息");
                    System.out.println("1.请输入员工编号");
                    String emNumber = scanner.next();
                    System.out.println("2.请输入员工姓名");
                    String emName = scanner.next();
                    System.out.println("请输入员工年龄");
                    int emAge = scanner.nextInt();
                    com.nny.demo513.Emp addemp = new com.nny.demo513.Emp(emNumber, emName, emAge);
                    testSet.addEmp(empSet, addemp);
                    System.out.println(empSet);
                    break;

                case 2:
                    testSet.findAllEmp(empSet);
                    break;

                case 3:
                    System.out.println("请输入你需要查询的姓名");
                    com.nny.demo513.Emp findemp = testSet.findEmpName(empSet, scanner.next());
                    System.out.println(findemp);
                    break;

                case 4:
                    System.out.println("请输入你需要删除的员工编号");
                    String number = scanner.next();
                    testSet.deleteNumber(empSet, number);
                    break;

                case 5:
                    System.out.println("请输入你需要查询的员工编号");
                    String findNumber = scanner.next();
                    com.nny.demo513.Emp findEmp = testSet.findEmpNumber(empSet, findNumber);
                    System.out.println(findEmp);
                    break;

                case 6:
                    System.out.println("请输入你需要修改的员工编号");
                    String updateNumber = scanner.next();
                    Emp updateEmp = testSet.findEmpNumber(empSet, updateNumber);
                    if (updateEmp != null) {
                        System.out.println("请输入你需要修改的员工年龄");
                        int updateAge = scanner.nextInt();
                        updateEmp.setAge(updateAge);
                        testSet.updateEmpAge(empSet, updateEmp);
                    }
        }
    }
    }}
